from .user import UserTools
from .dept import DeptTools
from .product import ProductTools
from .product_plan import ProductPlanTools
from .project import ProjectTools
from .task import TaskTools
from .bug import BugTools

# 导出所有API工具类，方便一次性导入
__all__ = ['UserTools', 'DeptTools', 'ProductTools', 'ProductPlanTools', 
           'ProjectTools', 'TaskTools', 'BugTools']

# 注册所有API工具的函数
def register_all_tools(mcp_instance, http_client):
    """
    注册所有API工具到MCP实例
    
    Args:
        mcp_instance: FastMCP实例
        http_client: HTTP客户端实例
    """
    # 用户工具
    user_tools = UserTools(http_client)
    user_tools.register(mcp_instance)
    
    # 部门工具
    dept_tools = DeptTools(http_client)
    dept_tools.register(mcp_instance)
    
    # 产品工具
    product_tools = ProductTools(http_client)
    product_tools.register(mcp_instance)
    
    # 产品计划工具
    product_plan_tools = ProductPlanTools(http_client)
    product_plan_tools.register(mcp_instance)
    
    # 项目工具
    project_tools = ProjectTools(http_client)
    project_tools.register(mcp_instance)
    
    # 任务工具
    task_tools = TaskTools(http_client)
    task_tools.register(mcp_instance)
    
    # Bug工具
    bug_tools = BugTools(http_client)
    bug_tools.register(mcp_instance)
    
    # 在这里可以添加更多工具注册...
    
    return {
        'user': user_tools,
        'dept': dept_tools,
        'product': product_tools,
        'product_plan': product_plan_tools,
        'project': project_tools,
        'task': task_tools,
        'bug': bug_tools,
        # 可以添加更多工具实例...
    } 